home *** CD-ROM | disk | FTP | other *** search
/ EuroCD 3 / EuroCD 3.iso / Programming / Python-1.4 / Lib / shutil.py < prev    next >
Encoding:
Python Source  |  1998-06-24  |  1.6 KB  |  70 lines
  1. # Module 'shutil' -- utility functions usable in a shell-like program
  2.  
  3. import os
  4.  
  5. MODEBITS = 010000    # Lower 12 mode bits
  6. # Change this to 01000 (9 mode bits) to avoid copying setuid etc.
  7.  
  8. # Copy data from src to dst
  9. #
  10. def copyfile(src, dst):
  11.     fsrc = open(src, 'r')
  12.     fdst = open(dst, 'w')
  13.     while 1:
  14.         buf = fsrc.read(16*1024)
  15.         if not buf: break
  16.         fdst.write(buf)
  17.  
  18. # Copy mode bits from src to dst
  19. #
  20. def copymode(src, dst):
  21.     st = os.stat(src)
  22.     mode = divmod(st[0], MODEBITS)[1]
  23.     os.chmod(dst, mode)
  24.  
  25. # Copy all stat info (mode bits, atime and mtime) from src to dst
  26. #
  27. def copystat(src, dst):
  28.     st = os.stat(src)
  29.     mode = divmod(st[0], MODEBITS)[1]
  30.     os.chmod(dst, mode)
  31.     os.utime(dst, st[7:9])
  32.  
  33. # Copy data and mode bits ("cp src dst")
  34. #
  35. def copy(src, dst):
  36.     copyfile(src, dst)
  37.     copymode(src, dst)
  38.  
  39. # Copy data and all stat info ("cp -p src dst")
  40. #
  41. def copy2(src, dst):
  42.     copyfile(src, dst)
  43.     copystat(src, dst)
  44.  
  45. # Recursively copy a directory tree.
  46. # The destination must not already exist.
  47. #
  48. def copytree(src, dst):
  49.     names = os.listdir(src)
  50.     os.mkdir(dst, 0777)
  51.     dot_dotdot = (os.curdir, os.pardir)
  52.     for name in names:
  53.         if name not in dot_dotdot:
  54.             srcname = os.path.join(src, name)
  55.             dstname = os.path.join(dst, name)
  56.             #print 'Copying', srcname, 'to', dstname
  57.             try:
  58.                 #if os.path.islink(srcname):
  59.                 #    linkto = os.readlink(srcname)
  60.                 #    os.symlink(linkto, dstname)
  61.                 #elif os.path.isdir(srcname):
  62.                 if os.path.isdir(srcname):
  63.                     copytree(srcname, dstname)
  64.                 else:
  65.                     copy2(srcname, dstname)
  66.                 # XXX What about devices, sockets etc.?
  67.             except os.error, why:
  68.                 print 'Could not copy', srcname, 'to', dstname,
  69.                 print '(', why[1], ')'
  70.